#pragma once

#include <iostream>
#include <cstdarg>
#include <string>
#include <ctime>
#include <sys/types.h>
#include <unistd.h>
#include <sys/stat.h>
#include <fcntl.h>

#include <unistd.h>
using namespace std;

// 日志等级
enum
{
    Debug,
    Info,
    Warning,
    Error,
    Fatal
};

// 日志文件类型
enum
{
    Screen,   // 显示器文件
    OneFile,  // 向一个文件打印
    ClassFile // 向多个文件打印
};

// 将日志等级转为字符串
string LevelToString(int level)
{
    switch (level)
    {
    case Debug:
        return "Debug";
    case Info:
        return "Info";
    case Warning:
        return "Warning";
    case Error:
        return "Error";
    case Fatal:
        return "Fatal";
    default:
        return "Unknown";
    }
}

const int default_style = Screen;       // 默认文件类型
const string default_filename = "log."; // 默认文件名
const string logdir = "log";            // 形成的日志默认所在路径

class Log
{
public:
    Log()
        : style(default_style), filename(default_filename)
    {
        mkdir(logdir.c_str(), 0775); // 当前路径下创建目录
    }

    // 用户自己决定是向显示器打还是向一个文件打还是向多个文件打
    void Enable(int sty)
    {
        style = sty;
    }

    // 将时间戳转化为年月日时分秒格式
    string TimeStampEXLocalTime()
    {
        time_t currtime = time(nullptr); // 获取当前时间戳
        struct tm *curr = localtime(&currtime);
        char time_buffer[128];
        snprintf(time_buffer, sizeof(time_buffer), "%d-%d-%d %d:%d:%d",
                 curr->tm_year + 1900, curr->tm_mon + 1, curr->tm_mday,
                 curr->tm_hour, curr->tm_min, curr->tm_sec);
        return time_buffer;
    }

    // 向一个日志文件写入
    void WriteLogToOneFile(const string &logname, const string &message)
    {
        int fd = open(logname.c_str(), O_CREAT | O_WRONLY | O_APPEND, 0666);
        if (fd < 0)
            return;
        write(fd, message.c_str(), message.size());
        close(fd);
    }

    // 向多个日志文件写入
    void WriteLogToClassFile(const string &levelstr, const string &message)
    {
        string logname = logdir;
        logname += "/";
        logname += filename;
        logname += levelstr;
        WriteLogToOneFile(logname, message);
    }

    // 向日志文件写入
    void WriteLog(const string &levelstr, const string &message)
    {
        switch (style)
        {
        case Screen:
            cout << message;
            break;
        case OneFile:
            WriteLogToClassFile("all", message);
            break;
        case ClassFile:
            WriteLogToClassFile(levelstr, message);
            break;
        default:
            break;
        }
    }

    void LogMessage(int level, const char *format, ...) // 可变参数列表
    {
        // 消息左半部分
        char leftbuffer[1024];
        string levelstr = LevelToString(level);   // 日志等级
        string currtime = TimeStampEXLocalTime(); // 当前时间
        string idstr = to_string(getpid());       // 进程id
        snprintf(leftbuffer, sizeof(leftbuffer), "[%s][%s][%s] ", levelstr.c_str(), currtime.c_str(), idstr.c_str());

        va_list args;
        va_start(args, format);
        char rightbuffer[1024];
        vsnprintf(rightbuffer, sizeof(rightbuffer), format, args);
        va_end(args); // args = nullptr

        string loginfo = leftbuffer;
        loginfo += rightbuffer;
        WriteLog(levelstr, loginfo);
    }

    ~Log()
    {
    }

private:
    int style;       // 文件类型
    string filename; // 文件名
};

Log lg;

class Conf
{
public:
    Conf()
    {
        lg.Enable(ClassFile);
    }
    ~Conf()
    {
    }
};

Conf conf;